# 题目三: 使用课程对应演示图片，首先计算灰度直方图，进一步使用大津算法进行分割，并比较分析分割结果。
import cv2 as cv
import matplotlib.pyplot as plt

# filename = r"D:\AI\Picture\mini.png"
# filename = r"D:\AI\Picture\pic2.png"
filename = r"D:\AI\Picture\pic6.png"
img = cv.imread(filename)
# 计算灰度直方图并显示
gray = cv.cvtColor(img, cv.COLOR_BGR2GRAY)  # 转换为灰度图
plt.hist(gray.ravel(), 256, [0, 256])  # 计算灰度直方图
plt.show()

# 先转灰度图，然后使用大津算法进行分割
gray = cv.cvtColor(img, cv.COLOR_BGR2GRAY)  # 转换为灰度图
_, dst = cv.threshold(gray, 0, 255, cv.THRESH_OTSU)  # 使用大津算法进行图像二值化
cv.imshow("source image", img)
cv.imshow("thresh_otsu img", dst)

cv.waitKey()
cv.destroyAllWindows()
